# +----------------------------------------------------------------------
# | DjangoAdmin敏捷开发框架 [ 赋能开发者，助力企业发展 ]
# +----------------------------------------------------------------------
# | 版权所有 2021~2024 北京DjangoAdmin研发中心
# +----------------------------------------------------------------------
# | Licensed LGPL-3.0 DjangoAdmin并不是自由软件，未经许可禁止去掉相关版权
# +----------------------------------------------------------------------
# | 官方网站: https://www.djangoadmin.cn
# +----------------------------------------------------------------------
# | 作者: @一米阳光 团队荣誉出品
# +----------------------------------------------------------------------
# | 版权和免责声明:
# | 本团队对该软件框架产品拥有知识产权（包括但不限于商标权、专利权、著作权、商业秘密等）
# | 均受到相关法律法规的保护，任何个人、组织和单位不得在未经本团队书面授权的情况下对所授权
# | 软件框架产品本身申请相关的知识产权，禁止用于任何违法、侵害他人合法权益等恶意的行为，禁
# | 止用于任何违反我国法律法规的一切项目研发，任何个人、组织和单位用于项目研发而产生的任何
# | 意外、疏忽、合约毁坏、诽谤、版权或知识产权侵犯及其造成的损失 (包括但不限于直接、间接、
# | 附带或衍生的损失等)，本团队不承担任何法律责任，本软件框架禁止任何单位和个人、组织用于
# | 任何违法、侵害他人合法利益等恶意的行为，如有发现违规、违法的犯罪行为，本团队将无条件配
# | 合公安机关调查取证同时保留一切以法律手段起诉的权利，本软件框架只能用于公司和个人内部的
# | 法律所允许的合法合规的软件产品研发，详细声明内容请阅读《框架免责声明》附件；
# +----------------------------------------------------------------------

import datetime

from fastapi.logger import logger
from sqlalchemy import and_
from werkzeug.datastructures.structures import MultiDict

from apps.models.user import User
from apps.models.data import Data
from apps.models.data_goods import DataGoods
from apps.models.data_order import DataOrder
from apps.forms.data_order import ApiDataOrderForm
from extends import db
from utils import R, regular
from apps.constants.message import PAGE_LIMIT
from utils.utils import uid


# 查询分页数据
async def DataOrderList(request):
    try:
        ID = request.query_params.get('id', 0)
        dataRow = db.query(DataGoods).filter(DataGoods.id == ID).first()
        if not dataRow:
            return R.failed("参数错误")
        # 页码
        page = int(request.query_params.get("page", 1))
        # 每页数
        limit = int(request.query_params.get("limit", PAGE_LIMIT))
        # 实例化查询对象
        query = db.query(DataOrder).filter(and_(DataOrder.is_delete == 0, DataOrder.shop_num == dataRow.shop_num,
                                                DataOrder.goods_id == dataRow.goods_id))
        userID = uid(request)
        if userID != 1:  # 不是超级管理员
            user = db.query(User).filter(User.id == userID).first()
            query = query.filter(DataOrder.company_id == user.company_id)
        # 排序
        query = query.order_by(DataOrder.order_time.desc())
        # 记录总数
        count = query.count()
        # 分页查询
        list = query.limit(limit).offset((page - 1) * limit).all()
        # 实例化数组对象
        result = []
        # 遍历数据源
        if list:
            for item in list:
                # 对象转字典
                data = item.to_dict()
                # 加入列表
                result.append(data)
        # 返回结果
        return R.ok(data=result, count=count)
    except:
        # 抛出异常
        logger.error('运行异常')
        raise
    finally:
        # 关闭连接
        db.close()


# 添加数据
async def DataOrderAdd(request):
    # 获取请求参数
    json_data = await request.json()
    # 表单验证
    form = ApiDataOrderForm(MultiDict(json_data))
    if not form.validate():
        # 获取错误描述
        err_msg = regular.get_err(form)
        # 返回错误信息
        return R.failed(msg=err_msg)
    # 根据手机号查询用户记录
    user = db.query(User).filter(and_(User.mobile == form.mobile.data, User.is_delete == 0)).first()
    # 查询结果判空
    if not user:
        return R.failed("用户不存在")
    del form.mobile
    # 判断店铺数据是否存在
    today = datetime.date.today()
    yesterday = today - datetime.timedelta(days=1)
    row = db.query(DataOrder).filter(DataOrder.order_sn == form.order_sn.data).first()
    if not row:  # 不存在新增
        form.day.data = yesterday
        addData = form.data
        data = DataOrder(**addData)
        data.uid = user.id
        if user.company_id:
            data.company_id = user.company_id
        # 插入数据
        data.save()
        # 返回结果
        return R.ok(msg="添加成功")
    else:
        return R.failed("数据已存在")
